# -*- coding: utf-8 -*-
import logging
import os
from datetime import datetime


class Logger:
    def __init__(self, log_dir, log_name):
        """
        初始化日志记录器
        :param log_dir: 日志文件目录
        :param log_name: 日志文件名
        """
        self.log_dir = log_dir
        self.log_name = log_name
        self.logger = self._get_logger()

    def _get_logger(self):
        """
        配置日志记录器
        :return: logger对象
        """
        # 创建logger对象
        logger = logging.getLogger(self.log_name)
        logger.setLevel(logging.INFO)

        # 创建日志目录
        if not os.path.exists(self.log_dir):
            os.makedirs(self.log_dir)

        # 创建文件处理器
        log_file = os.path.join(self.log_dir, 'log', f'{self.log_name}.log')
        file_handler = logging.FileHandler(log_file, encoding='utf-8')
        file_handler.setLevel(logging.INFO)

        # 创建控制台处理器
        console_handler = logging.StreamHandler()
        console_handler.setLevel(logging.INFO)

        # 设置日志格式
        formatter = logging.Formatter(
            '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
        )
        file_handler.setFormatter(formatter)
        console_handler.setFormatter(formatter)

        # 添加处理器到logger
        logger.addHandler(file_handler)
        logger.addHandler(console_handler)

        return logger

    def get_logger(self):
        """
        获取logger对象
        :return: logger对象
        """
        return self.logger